# ===== 完整的 PyTorch 修复代码 =====
# 将以下代码复制到你的 Jupyter notebook 的第一个代码单元格中

import os
import sys

# 设置环境变量禁用 TorchDynamo
os.environ['TORCH_DYNAMO_DISABLE'] = '1'
os.environ['TORCH_COMPILE_DISABLE'] = '1'
os.environ['PYTORCH_DISABLE_TORCH_COMPILE'] = '1'

# 导入 torch
import torch
print(f"PyTorch 版本: {torch.__version__}")

# 禁用 TorchDynamo
if hasattr(torch, '_dynamo'):
    torch._dynamo.config.suppress_errors = True
    torch._dynamo.config.disable = True
    print("✓ 已禁用 TorchDynamo")

# 禁用编译
if hasattr(torch, 'compile'):
    torch.compile = lambda *args, **kwargs: args[0] if args else lambda x: x
    print("✓ 已禁用 torch.compile")

# 导入其他模块
import torch.nn as nn
import torch.optim as optim
print("✓ PyTorch 模块导入成功！")

# 测试创建优化器
model = nn.Linear(10, 1)
optimizer = optim.Adam(model.parameters(), lr=0.001)
print("✓ 优化器创建成功！")

# ===== 修复完成 =====